import { ScanResult } from './scanner';
import { AIResult } from './ai';

/**
 * 基于规则的自动修复建议（示例：SQL注入）
 */
export function generateRuleFix(code: string, result: ScanResult): string | undefined {
  if (result.ruleId.includes('sql-injection')) {
    // 简单正则替换示例
    return code.replace(/(\w+)\s*=\s*input/g, '$1 = sanitize(input)');
  }
  // 可扩展更多规则
  return undefined;
}

/**
 * 基于AI的修复建议（直接返回AI建议）
 */
export function getAIFix(aiResult: AIResult): string | undefined {
  return aiResult.suggestion;
} 